from pages.blocks.movie_info import MovieInfo
from pages.internal_page import InternalPage
from selenium.webdriver.support.select import Select


class AddMoviePage(InternalPage):

    def __init__(self, driver, base_url):
        super(AddMoviePage, self).__init__(driver, base_url)
        self.movie_info = MovieInfo(self.driver, self.base_url)

    @property
    def imdbsearch(self):
        return self.driver.find_element_by_id("imdbsearch")


    @property
    def imdbsearch(self):
        return self.driver.find_element_by_css_selector("css=input[type='submit']")

